from django.db import models


# Create your models here.

class ChitCaseList(models.Model):
    type_choice = (
        (0, '向量库'),
        (1, '政策法规'),
        (2, '白皮书'),
    )
    cmd = models.CharField(verbose_name='指令', max_length=1280, null=True, blank=True)
    type = models.SmallIntegerField(verbose_name='类型', choices=type_choice, null=True, blank=True)
    result = models.CharField(verbose_name='一键审核结论', max_length=256, null=True, blank=True)
    poeple_result = models.CharField(verbose_name='人工复审结论', max_length=256, null=True, blank=True)
    source = models.CharField(verbose_name='出处', max_length=2048, null=True, blank=True)
    page = models.CharField(verbose_name='页码', max_length=256, null=True, blank=True)
    comment = models.CharField(verbose_name='备注', max_length=1280, null=True, blank=True)


class ChitCompareCase(models.Model):

    cmd = models.ForeignKey(verbose_name="指令", to='ChitCaseList', null=True, blank=True)
    type = models.SmallIntegerField(verbose_name='类型', null=True, blank=True)
    output = models.CharField(verbose_name='模型输出', max_length=256, null=True, blank=True)
    ans = models.CharField(verbose_name='标准答案', max_length=256, null=True, blank=True)
    poeple_result = models.CharField(verbose_name='人工复审结论', max_length=256, null=True, blank=True)
